from dataclasses import dataclass, field

@dataclass
class model_config:
    """配置其他模型参数"""
    model_path = r"/data/baidu/ERNIE-4.5-0.3B-PT" # 模型读取路径
    output_dir = r"./ernie_sft_lora_tuning_results"
    train_data_path = r"/data/train_class/data/trian_data/training_data.jsonl"
    test_data_path = "/data/train_class/data/test_data/test_data.jsonl"

@dataclass
class lora_config:
    """配置lora"""
    use_4bit_quant: bool = True
    r: int = 64
    lora_alpha: int = 32
    lora_dropout: float = 0.1
